package com.mediocre.customLogin.config;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.mediocre.common.util.WebUtils;
import com.mediocre.customLogin.filter.VerifyCodeFilter;
import com.mediocre.customLogin.service.UserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.security.access.hierarchicalroles.RoleHierarchy;
import org.springframework.security.access.hierarchicalroles.RoleHierarchyImpl;
import org.springframework.security.config.annotation.authentication.builders.AuthenticationManagerBuilder;
import org.springframework.security.config.annotation.web.builders.HttpSecurity;
import org.springframework.security.config.annotation.web.builders.WebSecurity;
import org.springframework.security.config.annotation.web.configuration.WebSecurityConfigurerAdapter;
import org.springframework.security.crypto.password.NoOpPasswordEncoder;
import org.springframework.security.crypto.password.PasswordEncoder;
import org.springframework.security.web.authentication.UsernamePasswordAuthenticationFilter;

import java.io.PrintWriter;

@Configuration
public class SecurityConfig extends WebSecurityConfigurerAdapter {

    @Autowired
    private UserService userService;

    @Autowired
    private VerifyCodeFilter verifyCodeFilter;

    @Bean
    PasswordEncoder passwordEncoder() {
        return NoOpPasswordEncoder.getInstance();
    }

    // 基于数据库配置测试用户
    @Override
    protected void configure(AuthenticationManagerBuilder auth) throws Exception {
        auth.userDetailsService(userService);
    }

    @Bean
    RoleHierarchy roleHierarchy() {
        RoleHierarchyImpl roleHierarchy = new RoleHierarchyImpl();
        roleHierarchy.setHierarchy("ROLE_admin > ROLE_user");
        return roleHierarchy;
    }

    @Override
    public void configure(WebSecurity web) throws Exception {
        web.ignoring().antMatchers("/index.html", "/js/**", "/css/**", "/images/**", "/img/**", "/fonts/**", "/favicon.ico", "/getVerifyCode");
    }

    @Override
    protected void configure(HttpSecurity http) throws Exception {
        http.addFilterBefore(verifyCodeFilter, UsernamePasswordAuthenticationFilter.class);
        http.authorizeRequests()
                /*
                这里的匹配规则我们采用了 Ant 风格的路径匹配符，Ant 风格的路径匹配符在 Spring 家族中使用非常广泛，它的匹配规则也非常简单：
                    通配符	含义
                    **	匹配多层路径
                    *	匹配一层路径
                    ?	匹配任意单个字符
                 */
                // 1 如果请求路径满足 /admin/** 格式，则用户需要具备 admin 角色
                .antMatchers("/admin/**").hasRole("admin")
                // 2 如果请求路径满足 /user/** 格式，则用户需要具备 user 角色。
                .antMatchers("/user/**").hasRole("user")
                // 3 剩余的其他格式的请求路径，只需要认证（登录）后就可以访问。(拦截规则的顺序不能写错)
                .anyRequest().authenticated()
                .and()
                .formLogin()
                // 配置登录页面和登录接口的地址
                //   默认的登录页面和登录接口的地址都是 /login
                .loginPage("/login.html")
                // 配置登录接口地址
                //   默认的登录接口地址就是 登录页面的地址
                .loginProcessingUrl("/doLogin")
                .usernameParameter("user")
                .passwordParameter("passwd")
                // 前后端分离时，登录成功返回json数据
                .successHandler((req, resp, auth) -> {
                    // auth 的类型是 Authentication，保存了我们刚刚登录成功的用户信息
                    Object principal = auth.getPrincipal();
                    WebUtils.out(resp, new ObjectMapper().writeValueAsString(principal));
                })
                // 前后端分离时，登录失败返回json数据
                .failureHandler((req, resp, e) -> {
                    // e的类型是AuthenticationException，对于登录失败，会有不同的原因，e 中则保存了登录失败的原因，我们可以将之通过 JSON 返回到前端
                    WebUtils.out(resp, e.getMessage());
                })
                .permitAll()
                // 前后端分离时，注销登录的处理方案
                .and()
                .logout()
                .logoutUrl("/logout")
                .logoutSuccessHandler((req, resp, auth) -> {
                    WebUtils.out(resp, "注销成功");
                })
                .permitAll()
                .and()
                .csrf().disable()
                // 前后端分离时，未认证处理方案
                .exceptionHandling()
                .authenticationEntryPoint((req, resp, authException) -> {
                    WebUtils.out(resp, "尚未登录，请先登录");
                });
    }
}
